Python 批量删除文件夹中指定类型的文件

例如：删除后缀为 .bmp 类型的文件

import sys, csv, operator
import os
import glob

# 删除指定类型的文件


if __name__ == '__main__':
    loadpath = 'K:\\LIDC-IDIR\\Linshi\\' # 数据集分根目录
    for indexA in os.listdir(loadpath):   #os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表
        # indexA 为数据集根目录下的第一层目录
        for indexB in os.listdir(os.path.join(loadpath, indexA)):
            if indexB[-3:] == 'pkl': continue  # 如果是 .pkl 文件则跳过
            for indexC in os.listdir(os.path.join(loadpath, indexA, indexB)):
                for indexD in os.listdir(os.path.join(loadpath, indexA, indexB, indexC)): # 到此应该进入到数据集的最后一层
                    for infile in glob.glob(os.path.join(loadpath, indexA, indexB, indexC,'*.bmp')):
                        os.remove(infile)